from pprint import pprint
from .aflowml.client import AFLOWmlAPI

if __name__ == '__main__':
    with open('test.poscar', 'r') as input_file:
        ml = AFLOWmlAPI()
        data = ml.get_prediction(input_file.read(), 'plmf')
        pprint(data)
